package com.gage.commons.ueditor.upload;

import java.io.IOException;
import java.io.InputStream;
import java.util.Arrays;
import java.util.List;
import java.util.Map;

import javax.servlet.http.HttpServletRequest;

import org.apache.commons.fileupload.disk.DiskFileItemFactory;
import org.apache.commons.fileupload.servlet.ServletFileUpload;
import org.springframework.web.multipart.MultipartFile;
import org.springframework.web.multipart.MultipartHttpServletRequest;
import org.springframework.web.multipart.MultipartResolver;
import org.springframework.web.multipart.commons.CommonsMultipartResolver;

import com.gage.commons.ueditor.PathFormat;
import com.gage.commons.ueditor.define.AppInfo;
import com.gage.commons.ueditor.define.BaseState;
import com.gage.commons.ueditor.define.FileType;
import com.gage.commons.ueditor.define.State;

public class BinaryUploader {

	public static final State save(HttpServletRequest request,
            Map<String, Object> conf) {
        boolean isAjaxUpload = request.getHeader( "X_Requested_With" ) != null;

        if (!ServletFileUpload.isMultipartContent(request)) {
            return new BaseState(false, AppInfo.NOT_MULTIPART_CONTENT);
        }

        ServletFileUpload upload = new ServletFileUpload(
                new DiskFileItemFactory());

        if ( isAjaxUpload ) {
            upload.setHeaderEncoding( "UTF-8" );
        }

        try {
        	MultipartResolver resolver = new CommonsMultipartResolver(request.getSession().getServletContext());
        	MultipartHttpServletRequest multipartRequest = resolver.resolveMultipart(request);
          //  MultipartHttpServletRequest multipartRequest = (MultipartHttpServletRequest) request;
            MultipartFile multipartFile = multipartRequest.getFile(conf.get("fieldName").toString());
			/*FileItemIterator iterator = upload.getItemIterator(request);

			while (iterator.hasNext()) {
				fileStream = iterator.next();

				if (!fileStream.isFormField())
					break;
				fileStream = null;
			}
		*/
			 if (multipartFile == null) {
	                return new BaseState(false, AppInfo.NOTFOUND_UPLOAD_DATA);
	            }

	            String savePath = (String) conf.get("savePath");
	            String originFileName = multipartFile.getOriginalFilename();
	            String suffix = FileType.getSuffixByFilename(originFileName);

	            originFileName = originFileName.substring(0,
	                    originFileName.length() - suffix.length());
	            savePath = savePath + suffix;

	            long maxSize = ((Long) conf.get("maxSize")).longValue();

	            if (!validType(suffix, (String[]) conf.get("allowFiles"))) {
	                return new BaseState(false, AppInfo.NOT_ALLOW_FILE_TYPE);
	            }

	            savePath = PathFormat.parse(savePath, originFileName);

	            String physicalPath = (String) conf.get("rootPath") + savePath;

	            InputStream is = multipartFile.getInputStream();
	            State storageState = StorageManager.saveFileByInputStream(is,
	                    physicalPath, maxSize);
	            is.close();

	            if (storageState.isSuccess()) {
	                storageState.putInfo("url", PathFormat.format(savePath));
	                storageState.putInfo("type", suffix);
	                storageState.putInfo("original", originFileName + suffix);
	            }

	            return storageState;
	        } catch (IOException e) {
	        }
	        return new BaseState(false, AppInfo.IO_ERROR);
	}

	private static boolean validType(String type, String[] allowTypes) {
		List<String> list = Arrays.asList(allowTypes);

		return list.contains(type);
	}
}
